package com.homework19;

import java.io.File;
import java.util.HashMap;
import java.util.Scanner;

public class demo01 {
    private static HashMap<String, Integer> map = new HashMap<>();
    public static void main(String[] args) {
        String s = new Scanner(System.in).nextLine();
        File file = new File(s);
        if(file.isDirectory()){
            countFile(file);
            System.out.println(map);
        }else{
            System.out.println("该目录不存在或不是一个文件夹");
        }
    }

    private static void countFile(File file) {
        File[] files = file.listFiles();
        for (File file1 : files) {
            if (file1.isFile()) {
                String fileName = file1.getName();
                String fileType = " ";
                if (fileName.lastIndexOf(".")!=-1) {
                    fileType = fileName.substring(fileName.lastIndexOf(".")+1);//分隔文件.后面的一位开始截取  截取到最后
                } else {
                    fileType = "未知文件";
                }
                setCount(fileType);
            }else {
                countFile(file1);
            }
        }
    }

    private static void setCount(String fileType) {
        if(!map.containsKey(fileType)){
            map.put(fileType,1);
        }else{
            int count = map.get(fileType);
            map.put(fileType,++count);

        }
    }
}
